#ifndef __NEARESTMEANCLASSIFIER_H__
#define __NEARESTMEANCLASSIFIER_H__

#include "Classifier.h"

#include <vector>

class NearestMeanClassifier : public Classifier
{
public:
  NearestMeanClassifier();
  virtual ~NearestMeanClassifier();

  virtual int test(const Sample& s) const;
  virtual void train(const DataSet& ds);

protected:
  // A vector where each element is the mean vector for some texture class
  // The mean is taken over a set of training samples
  std::vector<std::vector<float> > means_;
};

#endif